tensor([[[[[0.3100, 0.3100],
           [0.3100, 0.3100]],

          [[0.3100, 0.3100],
           [0.3100, 0.3100]]],


         [[[0.3100, 0.3100],
           [0.3100, 0.3100]],

          [[0.3100, 0.3100],
           [0.3100, 0.3100]]]],



        [[[[0.2000, 0.2000],
           [0.2000, 0.2000]],

          [[0.2000, 0.2000],
           [0.2000, 0.2000]]],


         [[[0.2000, 0.2000],
           [0.2000, 0.2000]],

          [[0.2000, 0.2000],
           [0.2000, 0.2000]]]],



        [[[[0.2200, 0.2200],
           [0.2200, 0.2200]],

          [[0.2200, 0.2200],
           [0.2200, 0.2200]]],


         [[[0.2200, 0.2200],
           [0.2200, 0.2200]],

          [[0.2200, 0.2200],
           [0.2200, 0.2200]]]],



        [[[[0.2700, 0.2700],
           [0.2700, 0.2700]],

          [[0.2700, 0.2700],
           [0.2700, 0.2700]]],


         [[[0.2700, 0.2700],
           [0.2700, 0.2700]],

          [[0.2700, 0.2700],
           [0.2700, 0.2700]]]]], grad_fn=<RoundBackward1>)